Python arrays and matrices manipulation